// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_CHILD_CHILD_SHARED_BITMAP_MANAGER_H_
#define CONTENT_CHILD_CHILD_SHARED_BITMAP_MANAGER_H_

#include <stdint.h>

#include <memory>

#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/memory/shared_memory.h"
#include "cc/resources/shared_bitmap_manager.h"
#include "content/common/render_message_filter.mojom.h"
#include "mojo/public/cpp/bindings/thread_safe_interface_ptr.h"

namespace content {

class SharedMemoryBitmap : public cc::SharedBitmap {
public:
    base::SharedMemory* shared_memory() { return shared_memory_; }

protected:
    SharedMemoryBitmap(uint8_t* pixels,
        const cc::SharedBitmapId& id,
        base::SharedMemory* shared_memory);

    base::SharedMemory* shared_memory_;
};

class ChildSharedBitmapManager : public cc::SharedBitmapManager {
public:
    explicit ChildSharedBitmapManager(
        const scoped_refptr<mojom::ThreadSafeRenderMessageFilterAssociatedPtr>&
            render_message_filter_ptr);
    ~ChildSharedBitmapManager() override;

    // cc::SharedBitmapManager implementation.
    std::unique_ptr<cc::SharedBitmap> AllocateSharedBitmap(
        const gfx::Size& size) override;
    std::unique_ptr<cc::SharedBitmap> GetSharedBitmapFromId(
        const gfx::Size&,
        const cc::SharedBitmapId&) override;

    std::unique_ptr<cc::SharedBitmap> GetBitmapForSharedMemory(
        base::SharedMemory* mem);

private:
    void NotifyAllocatedSharedBitmap(base::SharedMemory* memory,
        const cc::SharedBitmapId& id);

    scoped_refptr<mojom::ThreadSafeRenderMessageFilterAssociatedPtr>
        render_message_filter_ptr_;

    DISALLOW_COPY_AND_ASSIGN(ChildSharedBitmapManager);
};

} // namespace content

#endif // CONTENT_CHILD_CHILD_SHARED_BITMAP_MANAGER_H_
